package Alibaba;

import common.TreeNode;

import java.util.*;

/**
 * @author zhangmin
 * @create 2022-01-17 14:48
 */
public class zigzagLevelOrder103 {

    /**
     * 103. 二叉树的锯齿形层序遍历
     * 给你二叉树的根节点 root ，返回其节点值的 锯齿形层序遍历 。（即先从左往右，再从右往左进行下一层遍历，以此类推，层与层之间交替进行）。
     * 先层序遍历后将奇数行进行反转
     * */
    public List<List<Integer>> zigzagLevelOrder(TreeNode root) {
        if (root==null) return new LinkedList<>();
        Queue<TreeNode> queue=new LinkedList<>();
        List<List<Integer>> res=new LinkedList<>();
        queue.offer(root);
        while (!queue.isEmpty()){
            int levelNum=queue.size();
            List<Integer> list=new LinkedList<>();
            for (int i = 0; i < levelNum; i++) {
                TreeNode cur = queue.poll();
                if (cur.left!=null) queue.offer(cur.left);
                if (cur.right!=null) queue.offer(cur.right);
                list.add(cur.val);
            }
            res.add(list);
        }
        for (int i = 0; i < res.size(); i++) {
            if (i%2==1){
                Collections.reverse(res.get(i));
            }
        }
        return res;
    }
}
